#ifndef NUMBER_CLASSIFIER_H
#define NUMBER_CLASSIFIER_H

#include <opencv2/opencv.hpp>
#include <opencv2/ml.hpp>

class NumberClassifier {
public:
    NumberClassifier();
    bool loadModel(const std::string& model_path);
    int predict(const cv::Mat& roi);
    cv::Mat preprocessROI(const cv::Mat& roi);
    
private:
    cv::Ptr<cv::ml::SVM> svm_model_;
    bool model_loaded_;
    const int img_size_ = 28;
};

#endif